package com.sanrxteam.servlet;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.security.Principal;
import java.util.Collection;
import java.util.Enumeration;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.UUID;

import javax.servlet.AsyncContext;
import javax.servlet.DispatcherType;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletInputStream;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import javax.servlet.http.HttpUpgradeHandler;
import javax.servlet.http.Part;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadBase;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

public class UploadServlet extends HttpServlet {

	protected void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		String message = UploadServlet(request);
		request.setAttribute("message", message);
		request.getRequestDispatcher("/message.jsp").forward(request, response);
	}

	protected void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		doGet(request, response);

	}

	public String UploadServlet(HttpServletRequest request) {
		// 得到文件上传的保存目录，将上传的文件存放于WEB-INF目录下，不允许外界直接访问，保证上传文件的安全
		String savePath = this.getServletContext().getRealPath("/WEB-INF/upload");
		//上传时生成的保存目录
		String tempPath = this.getServletContext().getRealPath("/WEB-INF/temp");
		File tmpFile = new File(tempPath);
//		File file = new File(savePath);
//		if (!file.exists() && !file.isDirectory()) {
//			System.out.println(savePath + "目录不存在，需要创建");
//			// 创建目录
//			file.mkdir();
//		}
		if(!tmpFile.exists()) {
			//创建临时目录
			tmpFile.mkdirs();
		}
		//消息提醒
		String message = "";
		try {
			// 使用Apache文件上传组件处理文件上传步骤：
			// 1、创建一个DiskFileItemFactory工厂
			DiskFileItemFactory factory = new DiskFileItemFactory();
			//设置工厂缓存区的大小，当上传的文件大小超过缓存区的大小时，就会生成临时文件存放到指定的临时目录当中。
			factory.setSizeThreshold(1024*100);//设置缓存区的大小为100KB,如果不指定，那么缓存区的大小默认是10KB
			//设置上传时生成的临时文件的保存目录
			factory.setRepository(tmpFile);
			// 2、创建一个文件上传解析器
			ServletFileUpload upload = new ServletFileUpload(factory);
			
		
			/**
			 * ChangMQ267
			 * 获取文件大小
			 * 
			 */
//			public void update
			
			
			
			
			//监听文件上传进度
			upload.setProgressListener(new ProgressListener() {
				
				@Override
				public void update(long pBytesRead, long pContentLength, int arg2) {
					
					System.out.println("文件大小为：" + pContentLength + ",当前已处理：" + pBytesRead);
					
					/**
					 * 文件大小为：14608，当前处理：4096
					 */
					
				}
			});
			// 解决上传文件名的中文乱码
			upload.setHeaderEncoding("UTF-8");
			// 3、判断提交上来的数据是否是上传表单的数据
			// HttpServletRequest request = null;
			if (!ServletFileUpload.isMultipartContent(request)) {
				return null;
			}
			
			//设置单个文件上传的大小的最大值，
			upload.setFileSizeMax(1024*1024*15);//设置上传文件最大值fileMax=15MB
			//设置上传文件总量的最大值，最大值=同时上传的多个文件的大小的最大值的和
			upload.setSizeMax(1024*1024*30);//设置上传文件总和最大值Max=30MB
			/*
			 * 4、使用ServletFileUpload解析器解析上传数据，解析结果返回的是一个List<FileItem>集合，
			 * 每一个FileItem对应一个From表单的输入项
			 */
			List<FileItem> list = upload.parseRequest(request);
			for (FileItem item : list) {
				// 如果fileitem中封装的是普通输入项的数据
				if (item.isFormField()) {
					String name = item.getFieldName();
					// 解决普通输入项的数据的中文乱码问题
					String value = item.getString("UTF-8");
					// value = new String(value.getBytes("iso8859-1"),"UTF-8");
					System.out.println(name + "=" + value);
				} else {
					// 如果fileitem中封装的是上传文件
					// 得到上传的文件名称，
					String fileName = item.getName();
					System.out.println(fileName);
					if (fileName == null || fileName.trim().equals("")) {
						continue;
					}
					// 注意：不同的浏览器提交的文件名是不一样的，有些浏览器提交上来的文件名是带有路径的，如： c:\a\b\1.txt，而有些只是单纯的文件名，如：1.txt
					// 处理获取到的上传文件的文件名的路径部分，只保留文件名部分
					fileName = fileName.substring(fileName.lastIndexOf("\\") + 1);
					//得到上传文件的拓展名
					String fileExtName = fileName.substring(fileName.lastIndexOf(".")+1);
					//如果需要限制上传的文件类型，那么可以通过文件的拓展名来判断上传的文件类型是否合法
					System.out.println("上传的文件的拓展名是：" +  fileExtName);
					// 获取item中的上传文件的输入流
					InputStream in = item.getInputStream();
					//得到文件保存的名称
					String saveFileName = makeFileName(fileName);
					//得到文件的保存目录
					String realSavePath = makePath(saveFileName,savePath);
					// 创建一个文件输出流
					FileOutputStream out = new FileOutputStream(realSavePath + "\\" + saveFileName);
					// 创建一个缓存区
					byte buffer[] = new byte[1024];
					// 判断输入流中的数据是否已经读完的标识
					int len = 0;
					// 循环将输入流读入到缓存区当中，(len = in.read(buffer))>0就表示in里面还有数据
					while ((len = in.read(buffer)) > 0) {
						// 使用FileOutputStream输出流将缓存区的数据写入到指定的目录(savePath + "\\" + fileName)当中
						out.write(buffer, 0, len);
					}
					// 关闭输入流
					in.close();
					// 关闭输出流
					out.close();
					// 删除处理文件上传时生成的临时文件
					//item.delete();
					message = "文件上传成功！";
				}
			}
		}catch(FileUploadBase.FileSizeLimitExceededException e) {
			e.printStackTrace();
			message = "单个文件超出最大值！！！";
			System.out.println("1");
			return message;
		}catch(FileUploadBase.SizeLimitExceededException e) {
			e.printStackTrace();
			message = "上传的文件的总的大小超出限制的最大值！！！";
			return message;
		}catch (Exception e) {
			message = "文件上传失败！";
			e.printStackTrace();
			
		}
		return message;
	}
	
	/**
	 * @Method: makeFileName
	 * @Description: 生成上传文件的文件名，文件名以：uuid+"_"+文件的原始名称
	 * @param filename 文件的原始名称
	 * @return uuid+"_"+文件的原始名称
	 */
	private String makeFileName(String fileName) {
		//为防止文件覆盖的现象发生，要为上传的文件产生一个唯一的文件名
		return UUID.randomUUID().toString() + "_" + fileName;
	}
	/**
	 * 为防止一个目录下面出现太多文件，要使用hash算法打散存储
	 * @Method: makePath
	 * @param filename 文件名，要根据文件名生成存储目录
	 * @param savePath 文件存储路径
	 * @return 新的存储目录
	 */ 
	private String makePath(String fileName, String savePath) {
		int hashcode = fileName.hashCode();
		int dir1 = hashcode&0xf; //0--15
		int dir2 = (hashcode&0xf0)>>4; //0--15
		//构建新的保存目录
		String dir = savePath + "\\" +dir1 + "\\" + dir2; //upload\2\3  upload\3\5
		//File既可以代表文件也可以代表目录
		File file = new File(dir);
		//如果目录不存在
		if( !file.exists() ) {
			//创建目录
			file.mkdirs();
		}
		return dir;
	}



}
